import requests
import json

from utils.config import load_config


def ConnAI():
    config = load_config()
    api_key = config['model']['api_key']

    # API的URL
    api_url = config['model']['url']

    # 请求头，包含API密钥
    headers = {
        'Content-Type': 'application/json',
        'Authorization': f'Bearer {api_key}'
    }

    # 请求体，定义模型、消息和其他参数
    data = {
        'model': config['model']['model'],  # 使用的模型
        'messages': [
            {'role': 'system', 'content': '你是一个高级运维工程师，你需要对用户的输入进行判断并输出命令'},
            {'role': 'user', 'content': '"'}
        ],
        'max_tokens': 60  # 生成的最大token数
    }

    # 发送POST请求
    response = requests.post(api_url, headers=headers, data=json.dumps(data))

    # 检查响应状态
    if response.status_code == 200:
        # 解析响应JSON
        result = response.json()
        # 提取生成的文本
        generated_text = result['choices'][0]['message']['content']
        print(generated_text)
        return generated_text
    else:
        print(f'Error: {response.status_code}')
        print(response.text)
        return response.text
